from .A2M_task_new import LanguageModelingTaskWithControl
from .linear.transformer_lm import LinearTransformerLanguageModel
